package com.hb.core.pojo;

import cn.hutool.core.collection.CollUtil;
import lombok.Getter;
import lombok.Setter;

import java.util.List;

/**
 * 数据权限定义实体类
 * 需要判断数据权限的接口, 必须拥有该类型的属性
 *
 * @author zhaojk
 */
@Setter
@Getter
public class RoleDataAuthEntity {
    /**
     * 企业ID
     */
    private List<Long> orgIds;
    /**
     * 用户ID
     */
    private List<Long> userIds;

    // 设置数据权限的查询条件 start
    public void addUserId(Long... userId) {
        if (userIds == null) {
            userIds = CollUtil.list(false);
        }
        CollUtil.addAll(userIds, userId);
    }

    public void addUserId(List<Long> userIds) {
        if (this.userIds == null) {
            this.userIds = CollUtil.list(false);
        }
        this.userIds.addAll(userIds);
    }

    public void addOrgId(Long... orgId) {
        if (orgIds == null) {
            orgIds = CollUtil.list(false);
        }
        CollUtil.addAll(orgIds, orgId);
    }

    public void addOrgId(List<Long> orgIds) {
        if (this.orgIds == null) {
            this.orgIds = CollUtil.list(false);
        }
        this.orgIds.addAll(orgIds);
    }
    // 设置数据权限的查询条件 end
}
